Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Sockeye 2 heafield quantize pr2 #812

Merged

Conversation

mjdenkowski
Copy link
Contributor

@mjdenkowski mjdenkowski commented May 20, 2020

This is a housekeeping PR that follows #771 including:

  • Version/Changelog updates
  • Code formatting
  • Updated Docker builds for CPU (int8 inference) and GPU (Horovod and NCCL)
  • sockeye.quantize CLI for annotating a trained model with int8 scaling factors for fast loading/quantization

When this PR is merged into sockeye_2_heafield_quantize, that branch can be merged immediately into sockeye_2.

Pull Request Checklist

  • Changes are complete (if posting work-in-progress code, prefix your pull request title with '[WIP]'
    until you can check this box.
  • Unit tests pass (pytest)
  • System tests pass (pytest test/system)
  • Passed code style checking (./style-check.sh)
  • Updated major/minor version in sockeye/__init__.py. Major version bump if this is a backwards incompatible change.
  • Updated CHANGELOG.md

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.

Copy link
Contributor

@fhieber fhieber left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks!


log_sockeye_version(logger)

params_best = os.path.join(args.model, C.PARAMS_BEST_NAME)
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code below could also be moved into another function with only one string argument to avoid calling main functions in a library use case.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

That's a good idea.

@mjdenkowski mjdenkowski merged commit 50393fc into sockeye_2_heafield_quantize May 22, 2020
@mjdenkowski mjdenkowski deleted the sockeye_2_heafield_quantize_pr2 branch May 22, 2020 13:55
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants